from selenium import webdriver
from selenium.webdriver.chrome.service import Service
import ddddocr
from selenium.webdriver.common.by import By
import base64

service = Service(executable_path=r"E:\mypython2411\chromedriver.exe")
driver = webdriver.Chrome(service=service)
driver.get("https://login.cdpf.org.cn/uams/person.html?user_type=4")
driver.maximize_window()
# 提取base64编码的图片数据（修正拆分逻辑）
src = driver.find_element(By.XPATH, '//*[@id="code"]').get_attribute('src')
print(src)
# 拆分后取列表的第二个元素（索引1），即base64编码的字符串
base64_data = src.split(',', 1)[1]  # 关键修正点：取列表中的图片数据部分
# 下载图片（使用base64_data字符串）
with open('./数字图片验证码/中国残疾人服务平台.png', 'wb') as f:
    f.write(base64.b64decode(base64_data))
#
# # 识别数字
# ocr = ddddocr.DdddOcr()
# with open("./数字图片验证码/中国残疾人服务平台.png", 'rb') as f:
#     bytes_data = f.read()
#     result = ocr.classification(bytes_data)
#     print("验证码识别结果：", result)

img_bytes = base64.b64decode(base64_data)
ocr = ddddocr.DdddOcr()
result = ocr.classification(img_bytes)
print(result)
driver.quit()